package com.ksyun.ks3;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.URL;
import java.net.URLDecoder;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.io.Console;
import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.net.JarURLConnection;
import java.util.LinkedHashSet;
import java.util.Set;

import com.ksyun.ks3.demo.AbstractDemo;
/**
 * @author lijunwei[lijunwei@kingsoft.com]  
 * 
 * @date 2014年11月24日 下午3:14:29
 * 
 * @description demo运行器
 **/
public class DemoRunner {
	public static void main(String[] args) throws ClassNotFoundException,
			IOException, IllegalAccessException {
		run();
	}

	public static void run() throws ClassNotFoundException, IOException,
			IllegalAccessException {
		while (true) {
			List<Class<?>> clazzs = getClasses("com.ksyun.ks3.demo");
			Map<String, String> ops = new LinkedHashMap<String, String>();
			int j = 0;
			for (int i = 0; i < clazzs.size(); i++) {
				String op = clazzs.get(i).getName().replace("Demo", "")
						.substring(19);
				if (!op.equals("Abstract") && !op.contains(".")) {
					ops.put(String.valueOf(j), op);
					j++;
				}
			}
			System.out.println("请输入要执行的操作编号");
			for (Entry<String, String> entry : ops.entrySet()) {
				System.out.println(entry.getKey() + ":" + entry.getValue());
			}
			int key = System.in.read()-48;
			int avl = System.in.available();
			boolean add = true;
			for (int i = 0; i < avl; i++) {
				int m = System.in.read();
				if(m>='0'&&m<='9'&&add){
					key = 10*key+(m-48);
				}else{
				    add=false;	
				}
			}
			if (!ops.containsKey(String.valueOf( key))) {
				System.out.println("无效的编号");
			} else {
				Class<?> clazz = Class.forName("com.ksyun.ks3.demo."
						+ ops.get(String.valueOf( key)) + "Demo");
				Method[] methods = clazz.getDeclaredMethods();
				Map<String, Method> mths = new LinkedHashMap<String, Method>();
				for (int i = 0; i < methods.length; i++) {
					mths.put(String.valueOf(i), methods[i]);
				}
				System.out.println("请输入要执行的方法编号");
				for (Entry<String, Method> entry : mths.entrySet()) {
					System.out.println(entry.getKey() + ":" + entry.getValue());
				}
				 key = System.in.read()-48;
				 avl = System.in.available();
				 add = true;
				for (int i = 0; i < avl; i++) {
					int m = System.in.read();
					if(m>='0'&&m<='9'&&add){
						key = 10*key+(m-48);
					}else{
					    add=false;	
					}
				}
				if (!mths.containsKey(String.valueOf( key))) {
					System.out.println("无效的编号");
				} else {
					Method m = mths.get(String.valueOf( key));
					try{
					System.out.println(m.invoke(clazz.newInstance(),
							new Object[] {}));
					}catch(Throwable e){
						e.printStackTrace();
					}

				}
			}
			System.out
					.println("分割线-------------------------------------------------------------");
		}
	}

	public static List<Class<?>> getClasses(String packageName) {

		// 第一个class类的集合
		List<Class<?>> classes = new ArrayList<Class<?>>();
		// 是否循环迭代
		boolean recursive = true;
		// 获取包的名字 并进行替换
		String packageDirName = packageName.replace('.', '/');
		// 定义一个枚举的集合 并进行循环来处理这个目录下的things
		Enumeration<URL> dirs;
		try {
			dirs = Thread.currentThread().getContextClassLoader()
					.getResources(packageDirName);
			// 循环迭代下去
			while (dirs.hasMoreElements()) {
				// 获取下一个元素
				URL url = dirs.nextElement();
				// 得到协议的名称
				String protocol = url.getProtocol();
				// 如果是以文件的形式保存在服务器上
				if ("file".equals(protocol)) {
					// 获取包的物理路径
					String filePath = URLDecoder.decode(url.getFile(), "UTF-8");
					// 以文件的方式扫描整个包下的文件 并添加到集合中
					findAndAddClassesInPackageByFile(packageName, filePath,
							recursive, classes);
				} else if ("jar".equals(protocol)) {
					// 如果是jar包文件
					// 定义一个JarFile
					JarFile jar;
					try {
						// 获取jar
						jar = ((JarURLConnection) url.openConnection())
								.getJarFile();
						// 从此jar包 得到一个枚举类
						Enumeration<JarEntry> entries = jar.entries();
						// 同样的进行循环迭代
						while (entries.hasMoreElements()) {
							// 获取jar里的一个实体 可以是目录 和一些jar包里的其他文件 如META-INF等文件
							JarEntry entry = entries.nextElement();
							String name = entry.getName();
							// 如果是以/开头的
							if (name.charAt(0) == '/') {
								// 获取后面的字符串
								name = name.substring(1);
							}
							// 如果前半部分和定义的包名相同
							if (name.startsWith(packageDirName)) {
								int idx = name.lastIndexOf('/');
								// 如果以"/"结尾 是一个包
								if (idx != -1) {
									// 获取包名 把"/"替换成"."
									packageName = name.substring(0, idx)
											.replace('/', '.');
								}
								// 如果可以迭代下去 并且是一个包
								if ((idx != -1) || recursive) {
									// 如果是一个.class文件 而且不是目录
									if (name.endsWith(".class")
											&& !entry.isDirectory()) {
										// 去掉后面的".class" 获取真正的类名
										String className = name.substring(
												packageName.length() + 1,
												name.length() - 6);
										try {
											// 添加到classes
											classes.add(Class
													.forName(packageName + '.'
															+ className));
										} catch (ClassNotFoundException e) {
											e.printStackTrace();
										}
									}
								}
							}
						}
					} catch (IOException e) {
						e.printStackTrace();
					}
				}
			}
		} catch (IOException e) {
			e.printStackTrace();
		}

		return classes;
	}

	public static void findAndAddClassesInPackageByFile(String packageName,
			String packagePath, final boolean recursive, List<Class<?>> classes) {
		// 获取此包的目录 建立一个File
		File dir = new File(packagePath);
		// 如果不存在或者 也不是目录就直接返回
		if (!dir.exists() || !dir.isDirectory()) {
			return;
		}
		// 如果存在 就获取包下的所有文件 包括目录
		File[] dirfiles = dir.listFiles(new FileFilter() {
			// 自定义过滤规则 如果可以循环(包含子目录) 或则是以.class结尾的文件(编译好的java类文件)
			public boolean accept(File file) {
				return (recursive && file.isDirectory())
						|| (file.getName().endsWith(".class"));
			}
		});
		// 循环所有文件
		for (File file : dirfiles) {
			// 如果是目录 则继续扫描
			if (file.isDirectory()) {
				findAndAddClassesInPackageByFile(
						packageName + "." + file.getName(),
						file.getAbsolutePath(), recursive, classes);
			} else {
				// 如果是java类文件 去掉后面的.class 只留下类名
				String className = file.getName().substring(0,
						file.getName().length() - 6);
				try {
					// 添加到集合中去
					classes.add(Class.forName(packageName + '.' + className));
				} catch (ClassNotFoundException e) {
					e.printStackTrace();
				}
			}
		}
	}
}
